A = [[1, 2, 3],[4, 5, 6]]   #为A赋值
B = [[1, 0, 0],[0, 1, 0]]   #为B赋值
A_plus_B = [[A[i][j] + B[i][j]  for j in range(len(A[0]))]for i in range(len(A))]   #以A的第0个元素的长度为外层遍历为i，以A的长度为内层遍历为j，每轮把A[i][j]和B[i][j]相加作为新的列表
'''变形为
t=[]
A_plus_B=[]
for i in range(len(A)):
    t=[]
    for j in range(len(A[0])):
        t.append(A[i][j]+B[i][j])

    A_plus_B.append(t)
'''
print(A_plus_B)